package com.leecode.section;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;

/**
 * <p>
 * 拥有最多糖果的孩子
 *
 * 给你一个数组 candies 和一个整数 extraCandies ，其中 candies[i] 代表第 i 个孩子拥有的糖果数目。
 * 对每一个孩子，检查是否存在一种方案，将额外的 extraCandies 个糖果分配给孩子们之后，此孩子有 最多 的糖果。注意，允许有多个孩子同时拥有 最多 的糖果数目。
 *
 * 示例 1：
 *
 * 输入：candies = [2,3,5,1,3], extraCandies = 3
 * 输出：[true,true,true,false,true]
 * 解释：
 * 孩子 1 有 2 个糖果，如果他得到所有额外的糖果（3个），那么他总共有 5 个糖果，他将成为拥有最多糖果的孩子。
 * 孩子 2 有 3 个糖果，如果他得到至少 2 个额外糖果，那么他将成为拥有最多糖果的孩子。
 * 孩子 3 有 5 个糖果，他已经是拥有最多糖果的孩子。
 * 孩子 4 有 1 个糖果，即使他得到所有额外的糖果，他也只有 4 个糖果，无法成为拥有糖果最多的孩子。
 * 孩子 5 有 3 个糖果，如果他得到至少 2 个额外糖果，那么他将成为拥有最多糖果的孩子。
 *
 * 示例 2：
 *
 * 输入：candies = [4,2,1,1,2], extraCandies = 1
 * 输出：[true,false,false,false,false]
 * 解释：只有 1 个额外糖果，所以不管额外糖果给谁，只有孩子 1 可以成为拥有糖果最多的孩子。
 * 示例 3：
 *
 * 输入：candies = [12,1,12], extraCandies = 10
 * 输出：[true,false,true]
 *
 * </p>
 *
 * @author qingshan
 * @since 2020-07-19 12:39
 */
public class MaximumCandy05 {

    public static void main(String[] args) {
        System.out.println(Arrays.toString(getMax(new int[]{2,3,5,1,3}, 3)));
    }

    public static String[] getMax(int[] arr, int candy) {
        String[] result = new String[arr.length];
        int arrMax = 0;
        for (int candyNum: arr) {
            if (candyNum > arrMax) {
                arrMax = candyNum;
            }
        }
        for (int i = 0; i < arr.length; i++) {
            if (arr[i] + candy >= arrMax) {
                result[i] = "true";
            } else {
                result[i] = "false";
            }
        }
        return result;
    }

    /**
     * 优化上述方法
     */
    public static List<Boolean> getMax2(int[] arr, int candy) {

//        String[] result = new String[arr.length];

        int arrMax = 0;
        for (int candyNum: arr) {
            if (candyNum > arrMax) {
                arrMax = candyNum;
            }
        }
        List<Boolean> result = new ArrayList<>();
        for (int i = 0; i < arr.length; i++) {
            result.add(arr[i] + candy >= arrMax);
        }
        return result;
    }
}
